from langchain.prompts import PromptTemplate

# 不带参数
prompt = PromptTemplate(
    input_variables=[],
    template="给我一个很土但是听起来好养活的路人甲起个小名",
)
print(prompt.format())

# 单参数
prompt = PromptTemplate(
    input_variables=["对象"],
    template="给我一个很土但是听起来好养活的{对象}起个小名",
)
print(prompt.format(对象="小狗"))

# 多参数
prompt = PromptTemplate(
    input_variables=["什么样", "对象"],
    template="给我一个{什么样}的{对象}起个小名",
)
print(prompt.format(什么样="很土但是听起来很好养活", 对象=""))

# 简化使用
prompt = PromptTemplate.from_template("给我一个很土但是听起来好养活的{对象}起个小名")
print(prompt.format(对象="小猫"))